package cn.feizhu.aimarket.constant;

import lombok.Getter;

/**
 * 文件类型枚举
 */
@Getter
public enum FileTypeEnum {

    /**
     * 文本文件
     */
    TXT("txt", "文本文件", ".txt"),

    /**
     * Word文档
     */
    WORD("word", "Word文档", ".docx"),

    /**
     * PDF文档
     */
    PDF("pdf", "PDF文档", ".pdf"),

    /**
     * Markdown文档
     */
    MARKDOWN("md", "Markdown文档", ".md");

    /**
     * 类型代码
     */
    private final String code;

    /**
     * 类型描述
     */
    private final String description;

    /**
     * 文件扩展名
     */
    private final String extension;

    FileTypeEnum(String code, String description, String extension) {
        this.code = code;
        this.description = description;
        this.extension = extension;
    }

    /**
     * 根据代码获取枚举
     *
     * @param code 类型代码
     * @return 文件类型枚举
     */
    public static FileTypeEnum getByCode(String code) {
        if (code == null) {
            return null;
        }
        for (FileTypeEnum fileTypeEnum : FileTypeEnum.values()) {
            if (fileTypeEnum.getCode().equals(code)) {
                return fileTypeEnum;
            }
        }
        return null;
    }

    /**
     * 判断是否为有效的文件类型
     *
     * @param code 类型代码
     * @return 是否有效
     */
    public static boolean isValidFileType(String code) {
        return getByCode(code) != null;
    }
}